#!/usr/bin/python3
# -*- coding: utf-8 -*-
# File  : __init__.py
# Author: anyongjin
# Date  : 2020/10/13

from DistillBert.students.base import StudentModel
from DistillBert.students.onlstm import OnLstmStudent

stu_model_map = {
    'base': StudentModel,
    'onlstm': OnLstmStudent
}
